import { setupDrag } from './setupDrag';
import { CompanyTypeEnum } from '@lib/shared/enums/commonEnum';

interface ScriptOptions {
  identifier: string; // 脚本标识
}

const scriptElementsMap = new Map<string, string>();

function extractSQLBotId(input: string) {
  const regex = /sqlbot-[^\s"']+/;
  const match = input.match(regex);
  return match ? match[0] : null;
}

export function loadScript(scriptContent: string, options: ScriptOptions): Promise<void> {
  return new Promise((resolve, reject) => {
    const content = scriptContent.trim();
    if (scriptElementsMap.has(options.identifier)) return;

    // 处理IIFE格式
    if (content.startsWith('(function')) {
      const scriptId = extractSQLBotId(content);
      if (scriptId) {
        scriptElementsMap.set(options.identifier, scriptId);
      }
      // eslint-disable-next-line no-eval
      eval(content);
      setTimeout(() => {
        const button = document.querySelector('.sqlbot-assistant-chat-button');
        setupDrag(button as HTMLElement);
      }, 300); // 等待DOM渲染
      resolve();
      return;
    }

    // 处理<script>标签
    if (content.startsWith('<script')) {
      const div = document.createElement('div');
      div.innerHTML = content;
      const originalScript = div.querySelector('script');
      if (!originalScript) {
        reject(new Error('无效的script标签'));
        return;
      }
      const script = document.createElement('script');
      // 复制所有属性
      for (let i = 0; i < originalScript.attributes.length; i++) {
        const attr = originalScript.attributes[i];
        if (attr.name === 'id') {
          scriptElementsMap.set(options.identifier, attr.value);
        }
        script.setAttribute(attr.name, attr.value);
      }
      script.onload = () => {
        setTimeout(() => {
          const button = document.querySelector('.sqlbot-assistant-chat-button');
          setupDrag(button as HTMLElement);
        }, 300); // 等待DOM渲染
      };

      document.body.appendChild(script);
      resolve();
      return;
    }

    reject(new Error('不支持的脚本格式'));
  });
}

export function removeScript(identifier: string): void {
  const scriptId = scriptElementsMap.get(identifier);
  if (scriptId && identifier === CompanyTypeEnum.SQLBot) {
    // 清理全局单例标记
    const propName = `${scriptId}-state`;
    delete (window as any)[propName];
    if ((window as any).sqlbot_assistant_handler) {
      delete (window as any).sqlbot_assistant_handler;
    }
    // 删除页面上渲染的
    const floatingElements = document.querySelectorAll('[id^="sqlbot-"]');
    floatingElements.forEach((el) => {
      if (el.parentNode && !el.parentNode.isEqualNode(document.body) && !el.parentNode.isEqualNode(document.head)) {
        (el.parentNode as Element).remove();
      }
      el.remove();
    });
    scriptElementsMap.delete(identifier);
  }
}
